#include <catch2/catch.hpp>

#include "solution.hpp"

using namespace solution_006;

TEST_CASE("6. ZigZag Conversion", "[006]") {

CHECK(Solution().convert("AB", 1)              == "AB");
CHECK(Solution().convert("PAYPALISHIRING", 3)  == "PAHNAPLSIIGYIR");
CHECK(Solution().convert("PAYPALISHIRING", 4)  == "PINALSIGYAHRPI");
CHECK(Solution().convert("PAYPALISHIRING", 14) == "PAYPALISHIRING");
CHECK(Solution().convert("A", 1)               == "A");

}
